from llmtcg.chat.models.moonshot import MoonshotChatModel
from llmtcg.chat.model import ChatContext
from llmtcg.chat.tools.datetime import CurrentDatetime, Sleep
from llmtcg.chat.tools.printer import ConsolePrinter


model = MoonshotChatModel(model=MoonshotChatModel.Model.MOONSHOT_V1_8K)
if not model.validate():
    exit(1)

chat_ctx = ChatContext(model)
chat_ctx.register_tool(CurrentDatetime())
chat_ctx.register_tool(ConsolePrinter())
chat_ctx.system_prompt = """
你是一个有用的，具备专业训练并能够使用工具解决问题的助手。在调用工具之前，你必须也要告诉我调用工具
要达成什么样的目的。
"""
resp = chat_ctx.chat("每隔五秒钟打印当前时间，持续五次。")
print(resp)